Minimum Spanning Tree In Python